# 使用枚举类
'''

'''
# 当我们需要定义常量的时候，一个办法是用大写的变量，来定义，比如定义月份常量：
JAN = 1
FEB = 2
# ...
NOV = 11
DEC = 12
# 好处是简单，缺点是类型是 int, 并且仍然是变量
# 更好的方法是，使用枚举类型定义一个class类型，然后，每个常量都是class的一个唯一实例，python 提供了 enum 模块来实现这个功能：
from enum import Enum

Month = Enum('Month', ('Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec'))
print(Month) # <enum 'Month'>

# 这样我们就获得了 Month 类型的枚举类，可以直接使用 Month.Jan 来引用一个常量，或者枚举它的所有成员：
for name,member in Month.__members__.items():
    print(name, '=>', member, ',', member.value)
'''
输出：
Jan => Month.Jan , 1
Feb => Month.Feb , 2
Mar => Month.Mar , 3
Apr => Month.Apr , 4
May => Month.May , 5
Jun => Month.Jun , 6
Jul => Month.Jul , 7
Aug => Month.Aug , 8
Sep => Month.Sep , 9
Oct => Month.Oct , 10
Nov => Month.Nov , 11
Dec => Month.Dec , 12
'''

print('-' * 100)

# value 属性则是自动赋值给成员的 int 常量，默认从 1 开始计数
# 如果需要更精准地控制枚举类型，可以从 Enum 派生出自定义类
from enum import Enum, unique

@unique
class WeekDay(Enum):
    Sun = 0 # Sun的value被设定为0
    Mon = 1
    Tue = 2
    Wed = 3
    Thu = 4
    Fri = 5
    Sat = 6
# @unique 装饰器可以帮助我们检查保证没有重复的值

# 访问这些枚举类型可以有若干种方法：
print(WeekDay.Mon) # 输出：WeekDay.Mon
print(WeekDay.Fri) # 输出：WeekDay.Fri
for name,member in WeekDay.__members__.items():
    print(name, '=>', member, ',', member.value)
'''
输出：
Sun => WeekDay.Sun , 0
Mon => WeekDay.Mon , 1
Tue => WeekDay.Tue , 2
Wed => WeekDay.Wed , 3
Thu => WeekDay.Thu , 4
Fri => WeekDay.Fri , 5
Sat => WeekDay.Sat , 6
'''